# Copyright 2024-2025 The Alibaba Wan Team Authors. All rights reserved.
import os

os.environ["TOKENIZERS_PARALLELISM"] = "false"

from .wan_i2v_A14B import i2v_A14B  # noqa: E402
from .wan_s2v_14B import s2v_14B  # noqa: E402
from .wan_t2v_A14B import t2v_A14B  # noqa: E402
from .wan_ti2v_5B import ti2v_5B  # noqa: E402

WAN_CONFIGS = {
    "t2v-A14B": t2v_A14B,
    "i2v-A14B": i2v_A14B,
    "ti2v-5B": ti2v_5B,
    "s2v-14B": s2v_14B,
}

SIZE_CONFIGS = {
    "720*1280": (720, 1280),
    "1280*720": (1280, 720),
    "480*832": (480, 832),
    "832*480": (832, 480),
    "704*1280": (704, 1280),
    "1280*704": (1280, 704),
    "1024*704": (1024, 704),
    "704*1024": (704, 1024),
}

MAX_AREA_CONFIGS = {
    "720*1280": 720 * 1280,
    "1280*720": 1280 * 720,
    "480*832": 480 * 832,
    "832*480": 832 * 480,
    "704*1280": 704 * 1280,
    "1280*704": 1280 * 704,
    "1024*704": 1024 * 704,
    "704*1024": 704 * 1024,
}

SUPPORTED_SIZES = {
    "t2v-A14B": ("720*1280", "1280*720", "480*832", "832*480"),
    "i2v-A14B": ("720*1280", "1280*720", "480*832", "832*480"),
    "ti2v-5B": ("704*1280", "1280*704"),
    "s2v-14B": ("720*1280", "1280*720", "480*832", "832*480", "1024*704", "704*1024", "704*1280", "1280*704"),
}
